'''
1.创建套接字
2.获得服务器的ip.port
3.链接服务器
4.获取下载文件名字(发送接受数据)
5.将文件名字发送给服务器
6.接受文件中的数据
7.保存接收到的数据到一个文件中
8.关闭套接字
'''
import socket


def main():
    # 1.创建套接字
    tcp_client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    # 2.获得服务器的ip.port
    dest_ip = input("请输入下载的服务器ip:")

    dest_port = int(input("请输入下载的服务器port:"))
    # 3.链接服务器
    tcp_client_socket.connect((dest_ip, dest_port))
    # 4.获取下载文件名字(发送接受数据)
    download_file_name = input("请输入要下载文件的名字:")
    # 5.将文件名字发送给服务器
    tcp_client_socket.send(download_file_name.encode("utf-8"))
    # 6.接受文件中的数据
    recv_data = tcp_client_socket.recv(1024)
    if recv_data:
        # 7.保存接收到的数据到一个文件中
        with open("[新]" + download_file_name, "wb") as f:
            f.write(recv_data)
    # 8.关闭套接字
    tcp_client_socket.close()


if __name__ == "__main__":
    main()
